#include "trap.h"

int mul(int a, int b) {
    int c = a * b;
    return c;
    }

    int test_data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};
    int ans_data[] = {1, 4, 9, 16, 25, 36, 49, 64, 81, 100};

    #define NR_DATA (sizeof(test_data) / sizeof(test_data[0]))

    int main() {
        int i;
        for (i = 0; i < NR_DATA; i++) {
            int a = test_data[i];
            int b = test_data[i];
            nemu_assert(mul(a, b) == ans_data[i]);
            }
            return 0;
            }
          